#coding=utf-8

# 爬取  先这样吧。明天再改代码！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！！。
# url = 'http://tieba.baidu.com/p/2460150866'
# 前三页所有的图片

import requests
from urllib import urlretrieve#导入urlli模块
import re
import time
import os
def get_code(url):#先把代码下载到本地。
    try:
        pages=requests.get(url)
        print(pages)
        # print(pages.text)
        return  pages.text
    except Exception as e:
        print(e)
def get_link(o_code):#根据代码找出图片并保存在本地
    try:
        p_lin=r'src="(.+?\jpg)" pic_ext'#继续加强正则表达式的学习
        imge=re.compile(p_lin)
        imge_list=re.findall(imge,o_code)
        if os.path.exists('picture1.jpg'):
            if os.path.exists('picture10000.jpg'):
                num=20000
                for url in imge_list:
                    urlretrieve(url, './picture' + '%s.jpg' % num)  # 下载图片，要理解思路
                    num = num + 1
                    time.sleep(1)
                print("本次下载了%s张图片."%(num-20000))
            else:
                num = 10000
                for url in imge_list:
                   urlretrieve(url, './picture' + '%s.jpg' % num)  # 下载图片，要理解思路
                   num = num + 1
                   time.sleep(1)
                print("本次下载了%s张图片." % (num - 10000))
        else:
            num=1
            for url in imge_list:
                urlretrieve(url,'./picture'+'%s.jpg'%num)#下载图片，要理解思路
                num=num+1
                time.sleep(1)
            print("本次下载了%s张图片." % (num - 1))
    except Exception as e:
        print(e)
if __name__=="__main__":
    htmls=['http://tieba.baidu.com/p/2460150866?pn={}'.format(str(n)) for n in range(1,4)]
    for html in htmls:
        print(html)
        code_1 = get_code(html)
        get_link(code_1)
    print('图片处理完毕')

